from langchain_ollama import ChatOllama
llm = ChatOllama(
    model="deepseek-r1:8b",
    # model="gemma3:1b",
    base_url="http://localhost:11434",  # Ollama 服务地址
    temperature=0.7,  # 创造性程度
    num_predict=512,  # 最大生成长度
)

from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate

# 用户的询问
human_template = "{human_input}"
human_prompt = HumanMessagePromptTemplate.from_template(human_template)

# 将以上所有信息结合为一个聊天提示
chat_prompt = ChatPromptTemplate.from_messages([ human_prompt])

prompt = chat_prompt.format_prompt(human_input="唐代有个诗人叫鲁迅，他写了一首诗叫《悯农》，你能背诵下吗？").to_messages()
print('prompt:',prompt)
# 接收用户的询问，返回回答结果
# response = llm(prompt)
response = llm.invoke(prompt)

print('response:',response.content)

#response: 实际上，“唐代有个诗人叫鲁迅”这一说法存在误解。**鲁迅**是中国现代文学的重要作家（1881-1936），而著名的《悯农》诗作者是**白居易**，他是唐代伟大的现实主义诗人。
# 白居易创作了两首广为流传的《悯农》诗，其中一首尤为著名
# hallucination，《悯农》的作者不是白居易，是李绅